#!/usr/bin/env python
# # -*- coding: utf-8 -*-

"""
@File:      config.py.py
@Author:    Jim.Dai.Cn
@Date:      2024/1/30 08:42
@Desc:         
"""


import os


# os.environ['DASHSCOPE_API_KEY'] = 'sk-fdeb689ee1274dce9292aa124df453f0'

llm_cfg_72B = {
    'model': 'Qwen-72B-Chat',
    'model_server': 'http://192.168.202.202:3389/api/v1',
    'generate_cfg': {
        'top_p': 0.2
    },
    'max_ref_token':32000
}
llm_cfg_vl = {
    'model': 'qwen-vl-plus',
    'model_server': 'dashscope',
    'generate_cfg': {
        'top_p': 0.32
    }
}
llm_cfg_dashscope_qwen2_7B = {
    'model': 'qwen2-7b-instruct',
    'model_server': 'dashscope',
    'generate_cfg': {
        'top_p': 0.2
    }
}
llm_cfg_llama_cpp_qwen2_7B = {
    'model': 'qwen2-7b-instruct',
    'model_server': 'http://localhost:8080/v1',
    'generate_cfg': {
        'top_p': 0.0
    }
}
llm_cfg_local_yi_9B_chat = {
    'model': 'Yi-1.5-9B-Chat-16K',
    'model_server': 'http://localhost:8080/v1',
    'generate_cfg': {
        'top_p': 0.0
    }
}
llm_cfg_local_yi_9B = {
    'model': 'Yi-1.5-9B-32K',
    'model_server': 'http://localhost:8080/v1',
    'generate_cfg': {
        'top_p': 0.0
    }
}

dir_test_doc = 'round1_test_data/'
dir_workspace = 'workspace/round1_test_data/'
dir_results = 'results/'
str_result_name = 'DataWizards_金融规则长文本_result.json'

str_sys_prompt_date_tool = '你是一个擅长日期计算和Python编码以及执行的程序员。'

data_wizard_tools = ['unit_check','date_check','code_interpreter']
